from appium.webdriver.common.appiumby import AppiumBy
from selenium.webdriver.common.by import By

from weekend.wework.base.wework_page import WeWorkPage
from weekend.wework.page.search_page import SearchPage


class Category(WeWorkPage):
    def goto_search_page(self):
        # 点击搜索，前往搜索界面
        self.find_and_click(By.XPATH, "//*[@resource-id='com.tencent.wework:id/l6v']")
        return SearchPage(self.driver)

